package com.guyaogg.gulimall.seckill.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;

/**
 * @author 顾遥
 */
@Slf4j
@Configuration
public class MyRabbitConfig {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * @return 注入 json 序列化
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();

    }

    /**
     * 定制 RabbitTemplate
     * #@PostConstruct MyRabbitConfig 对象创建以后，执行该方法
     *  发送端确认
     *
     *  消费端确认（默认确认，确认完自动删除,批量确认，如果宕机，数据就会丢失
     *     使用手动确认 ,只要没有明确说明签收，消息就一直为 Unacked 状态，即使宕机也不会数据丢失，重新变为 ready 状态
     *     listener:
     *       direct:
     *         acknowledge-mode: manual
     *     channel.basicAck(deliveryTag,false); 签收；业务成功完成执行
     *     channel.basicNack(deliveryTag,false,true); 拒签，重新入队，业务失败
     */
    @PostConstruct
    public void initRabbitTemplate() {
        // 设置确认回调

        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * RabbitMQ 服务器收到消息
             * @param correlationData 当前消息的唯一 id
             * @param ack 消息是否成功收到 ,只要消息抵达 Broker 就会返回 true
             * @param cause 失败原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {

                // TODO MQ 消息可靠性 - 消息丢失(最重要）
                // 1.做好消息确认机制（publisher、consumer【手动 ack】
                // 2.每一个发送的消息，都在数据库做好记录，再定期处理流失的消息
                // 服务器收到
                log.debug("confirm..CorrelationData[{}] ===> ack[{}] ====> cause[{}]", correlationData, ack, cause);

            }
        });
        // 设置消息抵达队列的确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列，就会触发这个失败回调
             * @param message 投递失败的消息，详细信息
             * @param replyCode 回复码
             * @param replyText 回复文本内容
             * @param exchange 当时这个消息发给哪个交换机
             * @param routingKey 当时这个消息用哪个路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                // 报错误了，修改数据库当前消息状态 -> 错误
                log.warn("Fail Message[{}] ==> replyCode[{}] ==> replyText[{}] ==> exchange[{}] ==> routingKey[{}]",message,replyCode,replyText,exchange,routingKey);

            }
        });
    }
}
